Home
last modified time | relevance | path

Searched defs:update_indices (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/
H A Dinplace_ops.cc64 std::vector<xla::XlaOp> update_indices; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc538 std::vector<HloInstruction*> slice_indices, update_indices; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc1257 Value update_indices = BuildReshapeOp(builder, rewriter, update_indices_raw, in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier_test.cc5049 std::vector<HloInstruction*> slice_indices, update_indices; in TEST_F() local