| /external/tensorflow/tensorflow/python/autograph/lang/ |
| directives.py | 32 def set_element_type(entity, dtype, shape=UNSPECIFIED): function
|
| /external/tensorflow/tensorflow/python/autograph/converters/ |
| slices_test.py | 41 def_.directives[directives.set_element_type] = {
62 def_.directives[directives.set_element_type] = {
67 def_.directives[directives.set_element_type] = {
|
| directives_test.py | 37 directives.set_element_type(l, 'a', string_var)
44 d = def_.directives[directives.set_element_type]
51 directives.set_element_type(a, 1, shape=2)
57 d = def_.directives[directives.set_element_type]
|
| slices.py | 70 directives.set_element_type,
|
| lists.py | 110 directives.set_element_type,
163 directives.set_element_type,
168 directives.set_element_type,
|
| lists_test.py | 84 def_.directives[directives.set_element_type] = {
119 def_.directives[directives.set_element_type] = {
|
| directives.py | 100 if live_val is directives.set_element_type:
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| bad_rng_shape_validation_test.cc | 55 sans_layout.set_element_type(F32);
|
| client_library_test_base.cc | 334 subshape->set_element_type(BF16);
391 subshape->set_element_type(BF16);
626 subshape->set_element_type(BF16);
|
| /external/tensorflow/tensorflow/contrib/autograph/examples/benchmarks/ |
| cartpole_benchmark.py | 48 ag.set_element_type(rewards, tf.float32)
52 ag.set_element_type(reverse_discounted, tf.float32)
114 ag.set_element_type(discounted_rewards, tf.float32)
115 ag.set_element_type(step_counts, tf.int32)
131 ag.set_element_type(game_rewards, tf.float32)
188 ag.set_element_type(mean_steps_per_iteration, tf.int32)
|
| /external/tensorflow/tensorflow/python/autograph/ |
| __init__.py | 48 from tensorflow.python.autograph.lang.directives import set_element_type
74 'set_element_type',
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| bfloat16_conversion_folding.cc | 70 hlo->mutable_shape()->set_element_type(BF16);
230 ->set_element_type(BF16);
|
| bfloat16_normalization.cc | 97 convert->mutable_shape()->set_element_type(to);
105 hlo->mutable_shape()->set_element_type(to);
245 subshape->set_element_type(F32);
|
| convolution_group_converter.cc | 261 reduce_window_shape.set_element_type(F32);
263 convert_shape.set_element_type(F32);
305 convert_back_shape.set_element_type(activation->shape().element_type());
|
| stable_sort_expander.cc | 74 iota_shape.set_element_type(S32);
|
| bfloat16_propagation.cc | [all...] |
| hlo_parser.cc | [all...] |
| shape_inference.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape.cc | 25 set_element_type(shape_proto.element_type());
60 proto.set_element_type(element_type_);
|
| shape.h | 93 void set_element_type(PrimitiveType value) { element_type_ = value; } function in class:xla::Shape
|
| shape_util.cc | 229 shape->set_element_type(element_type);
239 result.set_element_type(TUPLE);
250 result.set_element_type(OPAQUE);
257 result.set_element_type(TOKEN);
765 new_shape.set_element_type(type);
[all...] |
| literal_util.cc | 54 subshape->set_element_type(
|
| literal_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| categorical_op.cc | 148 uniform_shape.set_element_type(xla::F32);
|
| while_op.cc | 158 body_output_shape.set_element_type(xla::TUPLE);
|
