| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_test.cc | 41 BaseEmbeddingLookupOpModel(std::initializer_list<int> index_shape, in BaseEmbeddingLookupOpModel()
90 HybridEmbeddingLookupOpModel(std::initializer_list<int> index_shape, in HybridEmbeddingLookupOpModel()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | get_item_op.cc | 37 const TensorShape& index_shape = ctx->InputShape(1); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_slice_ops.cc | 51 const TensorShape index_shape = ctx->InputShape("indices"); in Compile() local
|
| D | split_op.cc | 106 const TensorShape index_shape = ctx->InputShape(2); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 136 const Shape& index_shape = index_vector->shape(); in ExpandIndexVectorIntoOperandSpace() local
|
| D | gather_expander.cc | 114 const Shape& index_shape = index_vector->shape(); in ExpandIndexVectorIntoOperandSpace() local
|
| D | shape_inference.cc | 2653 for (const Shape& index_shape : start_index_shapes) { in InferDynamicSliceShape() local
2760 for (const Shape& index_shape : start_index_shapes) { in InferDynamicUpdateSliceShape() local
|
| D | hlo_alias_analysis_test.cc | 1068 Shape index_shape = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
|
| D | algebraic_simplifier.cc | 2592 const Shape& index_shape = gather->operand(1)->shape(); in HandleGather() local
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 771 TensorShape index_shape({0, 3}); in ProcessBatch() local
803 const TensorShape index_shape({batch.num_tasks(), 3}); in EmitIndexTensor() local
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 495 TfLiteIntArray* index_shape = TfLiteIntArrayCreate(2); in FastParseExampleLite() local
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference.cc | 146 const int32_t* __restrict index_shape, in GetReduceInnerSize()
|
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 145 const int32_t* __restrict index_shape, in GetReduceInnerSize()
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 145 const int32_t* __restrict index_shape, in GetReduceInnerSize()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 1067 auto index_shape = ShapeUtil::ChangeElementType(valid_slice->shape(), S32); in ExchangeHaloAndGetValidData() local
|
| D | spmd_partitioner.cc | 495 auto index_shape = ShapeUtil::ChangeElementType(shape, S32); in PadWithValue() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 1685 xla::Shape index_shape; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 2132 ArrayRef<int64_t> index_shape = index_ty.getShape(); in Verify() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 4073 auto index_shape = index_type.getShape().vec(); in fold() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 4752 BaseEmbeddingLookupOpModel(std::initializer_list<int> index_shape, in BaseEmbeddingLookupOpModel()
|